// Copyright (c) 2024, BlockProject 3D

//

// All rights reserved.

//

// Redistribution and use in source and binary forms, with or without modification,

// are permitted provided that the following conditions are met:

//

//     * Redistributions of source code must retain the above copyright notice,

//       this list of conditions and the following disclaimer.

//     * Redistributions in binary form must reproduce the above copyright notice,

//       this list of conditions and the following disclaimer in the documentation

//       and/or other materials provided with the distribution.

//     * Neither the name of BlockProject 3D nor the names of its contributors

//       may be used to endorse or promote products derived from this software

//       without specific prior written permission.

//

// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS

// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT

// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR

// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR

// CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,

// EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,

// PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR

// PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF

// LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING

// NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS

// SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

use crate::compiler::message::{Field, Message};

use crate::compiler::util::types::TypeMapper;

use crate::gen::base::map::TypePathMapper;

use crate::gen::base::message::{Templates, Utilities};

use crate::gen::base::message_common::generate_field_type_inline;

use crate::gen::base::Error;

use itertools::Itertools;

fn gen_field_from_bytes_impl<U: Utilities, T: TypeMapper>(

    field: &Field,

    templates: &Templates,

    type_path_map: &TypePathMapper<T>,

    function: &str,

) -> Result<String, Error> {

    let mut scope = templates.template.scope();

    scope.var("name", &field.name);

    let codec_template = templates.get(field.codec())
?0
;

    let msg_type = generate_field_type_inline::<U, T>(field, codec_template, type_path_map, "from_bytes")
?0
;

    if let Some(
header6
) = &field.header {

  Branch (47:12): [True: 6, False: 49]

  Branch (47:12): [True: 0, False: 0]

  Branch (47:12): [Folded - Ignored]

        scope.var("header_name", &header.name);

    }

    scope.var("type", msg_type);

    if field.header.is_some() && 
field.ty.is_union()6
 {

  Branch (51:8): [True: 6, False: 49]
  Branch (51:34): [True: 4, False: 2]

  Branch (51:8): [True: 0, False: 0]
  Branch (51:34): [True: 0, False: 0]

  Branch (51:8): [Folded - Ignored]
  Branch (51:34): [Folded - Ignored]

        Ok(scope.render(function, &["field_union"]).unwrap())

    } else if field.header.is_some() {

  Branch (53:15): [True: 2, False: 49]

  Branch (53:15): [True: 0, False: 0]

  Branch (53:15): [Folded - Ignored]

        Ok(scope.render(function, &["field_header"]).unwrap())

    } else if field.ty.is_message_reference() {

  Branch (55:15): [True: 3, False: 46]

  Branch (55:15): [True: 0, False: 0]

  Branch (55:15): [Folded - Ignored]

        Ok(scope.render(function, &["field_msg"]).unwrap())

    } else {

        Ok(scope.render(function, &["field"]).unwrap())

    }

}

pub fn generate_from_bytes_impl<U: Utilities, T: TypeMapper>(

    msg: &Message,

    templates: &Templates,

    type_path_map: &TypePathMapper<T>,

    function: &str,

) -> Result<String, Error> {

    let fields = msg

        .fields

        .iter()

        .map(|field| gen_field_from_bytes_impl::<U, T>(field, templates, type_path_map, function))

        .collect::<Result<Vec<String>, Error>>()
?0

        .join("");

    let field_names = msg

        .fields

        .iter()

        .map(|field| templates.template.scope().var("name", &field.name).render(function, &["field_name"]).unwrap())

        .join("");

    Ok(templates

        .template

        .scope()

        .var("fields", fields)

        .var("field_names", field_names)

        .render("", &[function])

        .unwrap())

}

pub fn generate<'variable, U: Utilities, T: TypeMapper>(

    mut templates: Templates<'_, 'variable>,

    msg: &'variable Message,

    type_path_map: &TypePathMapper<T>,

    function: &str,

) -> Result<String, Error> {

    templates.template.var("msg_name", &msg.name);

    generate_from_bytes_impl::<U, T>(msg, &templates, type_path_map, function)

}